package com.pf.utils;

import lombok.extern.slf4j.Slf4j;

import javax.xml.bind.DatatypeConverter;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

/**
 * 短链接生成工具类
 *
 * @author PanFei
 * @version 1.0.0
 * @createTime 2021/12/8 23:04
 */
@Slf4j
public class ShortUrlUtils {

    public static void main(String[] args) {
        String longUrl = "http://localhost:8080/user/get?id=1001";
        String[] shortUrls = generateShortUrl(longUrl);
        for (String url : shortUrls) {
            System.out.println(url);
        }
    }


    /**
     * 长链接生成短链接
     *
     * @param longUrl
     * @return
     */
    public static String[] generateShortUrl(String longUrl) {
        // 可以自定义生成 MD5 加密字符传前的混合 KEY
        String key = "short_url";
        // 要使用生成 URL 的字符
        String[] chars = new String[]{"a", "b", "c", "d", "e", "f", "g", "h",
                "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t",
                "u", "v", "w", "x", "y", "z", "0", "1", "2", "3", "4", "5",
                "6", "7", "8", "9", "A", "B", "C", "D", "E", "F", "G", "H",
                "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T",
                "U", "V", "W", "X", "Y", "Z"
        };

        // 将长网址md5生成32位签名串
        String sMD5EncryptResult = "";
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            md.update((key + longUrl).getBytes());
            byte[] digest = md.digest();
            sMD5EncryptResult = DatatypeConverter.printHexBinary(digest).toUpperCase();
        } catch (NoSuchAlgorithmException e) {
            log.error(">>>>> ShortUrlUtils generateShortUrl error!", e);
        }

        String[] resUrl = new String[4];
        // 将生成的32位签名串分为4段，每段8个字节
        for (int i = 0; i < 4; i++) {
            // 把加密字符按照 8 位一组 16 进制与 0x3FFFFFFF 进行位与运算
            String sTempSubString = sMD5EncryptResult.substring(i * 8, i * 8 + 8);
            // 这里需要使用 long 型来转换，因为 Inteper .parseInt() 只能处理 31 位 , 首位为符号位 , 如果不用 long ，则会越界
            // 先转换为16进制，然后用 0x3FFFFFFF 进行位与运算，目的是格式化截取前30位
            long lHexLong = 0x3FFFFFFF & Long.parseLong(sTempSubString, 16);
            String outChars = "";
            //循环获得每组6位的字符串
            for (int j = 0; j < 6; j++) {
                // 把得到的值与 0x0000003D 进行位与运算，取得字符数组 chars 索引(具体需要看chars数组的长度   以防下标溢出，注意起点为0)
                // 0x0000003D 代表什么意思？它的10进制是61,61代表charts数组长度62的0到61的坐标。
                // 0x0000003D 与 lHexLong 进行位与运算，就是格式化为6位，即61内的数字
                // 保证了index绝对是61以内的值
                long index = 0x0000003D & lHexLong;
                // 把取得的字符相加
                outChars += chars[(int) index];
                // 每次循环按位右移 5 位，因为30位的二进制，分6次循环，即每次右移5位
                lHexLong = lHexLong >> 5;
            }
            // 把字符串存入对应索引的输出数组
            resUrl[i] = outChars;
        }
        return resUrl;
    }
}
